re-adding normal test code