Fix bug with warning message in small sizes