fixes problem with fitting model for too small training population