Fix tests spec