updated test and removed unnessasary test