Fixed behaviour of test