move to 2 seperate tests