fixed bug with cnns