Fix small bug. Fix specs