Fixed copy-paste error - fixing tests