fix unit tests due to change in using pspec