Fixes #7. Very quick and simple.