Fix for first training sample