Fixed tests

Tests no longer passed once we used `load_secret`.
Merge pull request #32 from retr0h/working

Fixed tests