Added import to get rid of implicit conversion warning