small fix for big bug