Fix tests for previous