Fixed one failing test