Slight change to fix unit tests