fixed last test error