Fix #1

This is a simple fix for #1. :)