Not working test fixed