bugfix

Code was throwing an exception when an `int` was supplied to the class..