changing test accordingly