Fix specs for code removal