Proper fix for #4

Another glorious one-liner.